/external/tensorflow/tensorflow/core/framework/ |
D | tensor_test.cc | 547 auto flat_outer_dims = t.flat_outer_dims<float>(); in TEST_F() local 548 EXPECT_EQ(2, flat_outer_dims.dimension(0)); in TEST_F() 549 EXPECT_EQ(60, flat_outer_dims.dimension(1)); in TEST_F() 550 EXPECT_EQ(flat_outer_dims(0, 0), 0.01f); in TEST_F() 551 EXPECT_EQ(flat_outer_dims(1, 59), 0.02f); in TEST_F() 554 auto flat_outer_dims = t.flat_outer_dims<float, 3>(); in TEST_F() local 555 EXPECT_EQ(2, flat_outer_dims.dimension(0)); in TEST_F() 556 EXPECT_EQ(3, flat_outer_dims.dimension(1)); in TEST_F() 557 EXPECT_EQ(20, flat_outer_dims.dimension(2)); in TEST_F() 558 EXPECT_EQ(flat_outer_dims(0, 0, 0), 0.01f); in TEST_F() [all …]
|
D | tensor.h | 323 typename TTypes<T, NDIMS>::Tensor flat_outer_dims(); 424 typename TTypes<T, NDIMS>::ConstTensor flat_outer_dims() const; 733 typename TTypes<T, NDIMS>::Tensor Tensor::flat_outer_dims() { in flat_outer_dims() function 750 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_outer_dims() const { in flat_outer_dims() function
|
/external/tensorflow/tensorflow/core/kernels/ |
D | batch_util.cc | 47 parent->flat_outer_dims<T>().chip(index, 0) = element.flat<T>(); in HandleElementToSlice() 54 auto parent_as_matrix = parent->flat_outer_dims<string>(); in HandleElementToSlice() 69 auto parent_as_matrix = parent->flat_outer_dims<Variant>(); in HandleElementToSlice() 87 element->flat<T>() = parent.flat_outer_dims<T>().chip(index, 0); in HandleSliceToElement()
|
D | cwise_op_select.cc | 79 FastBoundsCheck(then->flat_outer_dims<T>().dimension(1), in ComputeBroadcasting() 105 func(ctx->eigen_device<Device>(), output->flat_outer_dims<T>(), in ComputeBroadcasting() 106 cond->vec<bool>(), then->flat_outer_dims<T>(), in ComputeBroadcasting() 107 else_->flat_outer_dims<T>()); in ComputeBroadcasting()
|
D | training_ops.cc | 775 auto var_flat = var.flat_outer_dims<T>(); in DoCompute() 776 auto accum_grad_flat = accum_grad.flat_outer_dims<T>(); in DoCompute() 777 auto accum_update_flat = accum_update.flat_outer_dims<T>(); in DoCompute() 778 auto grad_flat = grad.flat_outer_dims<T>(); in DoCompute() 958 auto var_flat = var.flat_outer_dims<T>(); in Compute() 959 auto grad_flat = grad.flat_outer_dims<T>(); in Compute() 1299 auto var_flat = var.flat_outer_dims<T>(); in Compute() 1300 auto accum_flat = accum.flat_outer_dims<T>(); in Compute() 1301 auto grad_flat = grad.flat_outer_dims<T>(); in Compute() 1450 auto var_flat = var.flat_outer_dims<T>(); in Compute() [all …]
|
D | random_shuffle_op.cc | 93 const auto input_mat = input.flat_outer_dims<T>(); in Compute() 94 auto output_mat = output->flat_outer_dims<T>(); in Compute()
|
D | segment_reduction_ops.cc | 89 auto input_flat = input.flat_outer_dims<T>(); in Compute() 112 auto output_flat = output->flat_outer_dims<T>(); in Compute() 282 auto output_flat = output->flat_outer_dims<T>(); in ComputeAsync() 489 auto output_flat = output->flat_outer_dims<T>(); in Compute() 654 auto input_flat = input.flat_outer_dims<T>(); in Compute() 685 auto output_flat = output->flat_outer_dims<T>(); in Compute() 1023 auto input_flat = input.flat_outer_dims<T>(); in Compute() 1060 auto output_flat = output->flat_outer_dims<T>(); in Compute()
|
D | sparse_conditional_accumulator.h | 264 auto sum_flat = sum_tensor->flat_outer_dims<T>(); in AddToAccumGradFunction() 265 auto accum_flat = accum_val_->flat_outer_dims<T>(); in AddToAccumGradFunction() 266 auto grad_flat = grad_val->flat_outer_dims<T>(); in AddToAccumGradFunction() 324 auto accum_flat = accum_val_->flat_outer_dims<T>(); in DivideAccumGradByCounter()
|
D | scatter_op.cc | 124 auto params_flat = params.flat_outer_dims<T>(); in DoCompute() 197 auto params_flat = params.flat_outer_dims<T>(); in DoCompute()
|
D | inplace_ops_functor_gpu.cu.cc | 47 auto Toutput = output->flat_outer_dims<T>(); in DoParallelConcatUpdate()
|
D | dynamic_stitch_op.cc | 175 const int slice_size = merged->flat_outer_dims<T>().dimension(1); in Compute() 239 auto merged_flat = merged->flat_outer_dims<T>(); in Compute()
|
D | dynamic_partition_op.cc | 132 out_flat.push_back(outputs[p]->flat_outer_dims<T>()); in Compute()
|
D | inplace_ops.cc | 38 auto Toutput = output->flat_outer_dims<T>(); in DoParallelConcatUpdate()
|
D | gather_nd_op.cc | 169 auto params_flat = params.flat_outer_dims<T, IXDIM + 1>(); \ in DoGatherNd()
|
D | queue_base.cc | 45 auto parent_as_matrix = parent.flat_outer_dims<T>(); in HandleSliceToElement()
|
D | barrier_ops.cc | 313 auto values_matrix = values.flat_outer_dims<T>(); in InsertOneLocked()
|
D | resource_variable_ops.cc | 599 auto params_flat = params->flat_outer_dims<T>(); in Compute()
|
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
D | stats_accumulator_ops.cc | 196 auto gradients = gradients_t->flat_outer_dims<float>(); in SerializeTensorAccumulatorToOutput() 204 auto hessians = hessians_t->flat_outer_dims<float>(); in SerializeTensorAccumulatorToOutput() 277 const auto& gradients = gradients_t.flat_outer_dims<float>(); in AddToTensorAccumulator() 279 const auto& hessians = hessians_t.flat_outer_dims<float>(); in AddToTensorAccumulator()
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | map_and_batch_dataset_op.cc | 229 auto output_t = output->flat_outer_dims<type>(); \ in CopyPartialBatch() 230 auto value_t = value.flat_outer_dims<type>(); \ in CopyPartialBatch()
|